# -*- coding: utf-8 -*-
"""
@Time    : 2025/3/17 14:49 
@Author  : ZhangShenao 
@File    : 3.TTS文本转语音.py 
@Desc    : 使用TTS模型,根据文本生成语音
"""

import os

import dotenv
from openai import OpenAI

# 加载环境变量
dotenv.load_dotenv()

# 创建OpenAI客户端
client = OpenAI(base_url=os.getenv("OPENAI_API_BASE"))

# 指定文件路径
speech_file_path = "./data/静夜思.mp3"

# 指定语言内容
text = "床前明月光，疑似地上霜。举头望明月，低头思故乡"

# 调用TTS API,生成音频文件
response = client.audio.speech.create(
    model="tts-1",  # 指定模型,可选tts-1和tts-1-hd
    voice="alloy",  # 指定音色,TTS模型支持多种音色,如alloy、ash等
    input=text,  # 指定文本内容
)

# 将音频文件保存到本地
response.stream_to_file(speech_file_path)
print("音频生成成功")
